#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
#include <pthread.h>

#define PORT 8888
#define MAX_BUFFER_SIZE 1024
#define MAX_CLIENTS 128

// 存储客户端信息的结构体
struct SockInfo
{
    int fd;                      // 通信套接字
    pthread_t tid;               // 线程ID
    struct sockaddr_in addr;     // 客户端地址信息
};

// 全局客户端信息数组
struct SockInfo infos[MAX_CLIENTS];

// 线程工作函数
void* working(void* arg)
{
    struct SockInfo* info = (struct SockInfo*)arg;

    while (1)
    {
        char buf[MAX_BUFFER_SIZE] = {0};
        // 接收数据
        int ret = read(info->fd, buf, sizeof(buf));
        if (ret == 0)
        {
            printf("客户端已关闭连接，fd = %d\n", info->fd);
            break;
        }
        else if (ret == -1)
        {
            perror("读取数据失败");
            break;
        }

        printf("客户端 (%s:%d) say: %s\n",
               inet_ntoa(info->addr.sin_addr),
               ntohs(info->addr.sin_port),
               buf);

        // 回显数据
        write(info->fd, buf, ret);
    }

    // 释放当前客户端的资源
    close(info->fd);
    info->fd = -1;  // 标记为可用
    pthread_exit(NULL);
}

int main()
{
    // 创建监听套接字
    int listenfd = socket(AF_INET, SOCK_STREAM, 0);
    if (listenfd == -1)
    {
        perror("socket");
        exit(1);
    }

    // 设置地址复用
    int opt = 1;
    setsockopt(listenfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));

    // 绑定地址和端口
    struct sockaddr_in addr;
    addr.sin_family = AF_INET;
    addr.sin_port = htons(PORT);
    addr.sin_addr.s_addr = INADDR_ANY;
    if (bind(listenfd, (struct sockaddr*)&addr, sizeof(addr)) == -1)
    {
        perror("bind");
        close(listenfd);
        exit(1);
    }

    // 设置监听
    if (listen(listenfd, MAX_CLIENTS) == -1)
    {
        perror("listen");
        close(listenfd);
        exit(1);
    }

    // 初始化客户端信息数组
    for (int i = 0; i < MAX_CLIENTS; ++i)
    {
        infos[i].fd = -1;
    }

    printf("服务器启动成功，监听端口：%d\n", PORT);

    while (1)
    {
        // 找到空闲的 SockInfo
        struct SockInfo* info = NULL;
        for (int i = 0; i < MAX_CLIENTS; ++i)
        {
            if (infos[i].fd == -1)
            {
                info = &infos[i];
                break;
            }
        }

        if (info == NULL)
        {
            // 没有空闲的 SockInfo，暂时拒绝连接
            printf("服务器连接数已满，拒绝新连接。\n");
            sleep(1);
            continue;
        }

        // 接受客户端连接
        socklen_t len = sizeof(info->addr);
        info->fd = accept(listenfd, (struct sockaddr*)&info->addr, &len);
        if (info->fd == -1)
        {
            perror("accept");
            continue;
        }

        printf("新客户端连接：%s:%d\n",
               inet_ntoa(info->addr.sin_addr),
               ntohs(info->addr.sin_port));

        // 创建线程处理客户端通信
        pthread_create(&info->tid, NULL, working, info);
        pthread_detach(info->tid);  // 分离线程，自动回收
    }

    // 关闭监听套接字
    close(listenfd);
    return 0;
}
